#ifndef KNN_H
#define KNN_H

#include <opencv/cv.h>
#include <data.h>

using namespace std;
using namespace cv;

class KNN
{
    Data *data;
    vector<float> query;

public:
    KNN();
    KNN(Data *data);
    KNN(Data *data, vector<float>  query);
    void setData(Data *data, vector<float>  query);
    void setQuery(vector<float> query);
    vector<int> getResult();
};

#endif // KNN_H
